package com.hernie.piggymall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @Author HuangJp
 * @Date 2022/5/4
 * @Description
 */
@Configuration
public class MyRabbitConfig {

    @Autowired
    RabbitTemplate rabbitTemplate;

    /**
     * 将rabbitmq消息中的对象转换为json格式（默认是java序列化机制）
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     *  服务器收到消息就回调
     *      1、spring.rabbitmq.publisher-confirm-type=correlated
     *      2、设置确认回调
     *
     *  消息正确抵达队列进行回调
     *      1、spring.rabbitmq.publisher-returns=true
     *         spring.rabbitmq.template.mandatory=true
     *      2、设置确认回调
     *
     *  消费端确认（保证每个消息被正确消费，此时才可以broker删除这个消息）
     *          spring.rabbitmq.listener.direct.acknowledge-mode=manual 手动签收
     *      1、默认是自动确认的，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     *          问题：收到很多消息，自动回复给服务器ack，只有一个消息处理成功，此时宕机，就会发生消息丢失
     *          解决：手动确认，只要没有明确告诉MQ消息被消费，没有ack，消息就会一直是unacked状态，即使消费者宕机，消息也不会丢失，会重新变为ready状态，直到有新的消费者连接进来就发给他
     *            签收消息：
     *                  channel.basicAck(deliveryTag, false)签收，业务成功完成就应该签收
     *                  channel.basicNack(deliveryTag, false, false)拒签，业务失败，拒收
     *
     * 防止消息丢失
     *  1、做好消息确认机制（publisher, consumer[手动ack])
     *  2、每一个发送的消息都在数据库库做好记录。定期将失败的消息再次发送一遍
     *
     */
    @PostConstruct  // MyRabbitConfig对象创建完毕之后，执行这个方法（即构造方法执行后执行该方法）
    public void initRabbitTemplate() {
        // 设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 只要消息抵达Broker，confirm就确认成功
             * @param correlationData   当前消息的唯一关联数据（这个是消息的唯一id）
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                // 服务器收到
                // 修改消息状态
            }
        });
        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就会触发这个失败回调
             * @param message   投递失败的消息详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange  消息发送时指定的交换机
             * @param routingKey    消息发送时指定的路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 报错误了。修改数据库当前消息的状态->错误
            }
        });
    }
}
